I spent my income tax refund on both wheels and crank.
I can feel the difference in the wheels but not in the crank.
Get the wheels.
I got the Crane Creek Volos Track wheels and they changed my Pista for the better more than any other single component (and I have no original parts except for the frame).