Glad to see another wagon
I would definitely recommend also looking into sway bars, steel brake lines, and a catch cans/aos. Also, get a protune when you can.
If you're looking at wheels soon, I would 100% recommend Konig. Great prices for rotary forged and super well made. If you're curious about what sizes fit wagons, I have a square 17x8 +40 and 225/45R17. The rears rub on big bumps because I'm on sti springs, but if you cut the inside of the fenders they fit perfectly.