Hi,
Thanks for saying you replaced the PTO shaft.
You didn't say the year of the Cub.
Most of what I wrote above is info for the original PTO shaft an older Cub would have.
IH changed the PTO shaft later, it uses a snap ring ahead of the rear ball bearing.
If you replaced the shaft with a new shaft, the shafts TM Tractor sells have the snap ring, there is no staking used.
I don't know where you bought the new PTO shaft, or if it has the snap ring.
If it has the snap ring, the shaft should not move forward and back.
Below is the listing, you can look at the pic. The shaft has a groove in it for the snap ring.
http://www.tmtractor.com/new/pt/514fp.htm Below is a listing for a new PTO rear bearing, it shows the snap ring.
http://www.tmtractor.com/new/pt/155fp.htm If your old PTO shaft didn't have the snap ring, and you reused the old bearing on the new shaft, you might not have had a snap ring to put on the new shaft. It looks like the snap ring doesn't come with the new shaft.
You can probably buy a snap ring at an auto parts store, NAPA or others.
You would need to remove the PTO clutch at the front of the PTO shaft to put the snap ring on.