I traded in my '03 WRX for my ZR. Mine was a 2.0L. Did you swap out your stock 2.0L for a 2.5L engine? My WRX had Cobb Stage 2
Personally, I don't really like the looks of the new '18 STi. But the engine is proven and as you probably know, has lots of aftermarket support.