If the shaft came up then the transmission is bad and has to be replaced. If you are going to fix this washer then you have to check the agitator coupling and the agitator itself.
Here are the break down diagrams for the GE washer
Model WPSQ3120T5WW
Gene.