If it's the 4 speed, I believe it is 40TE transmission. If you look at the attachments below, they provide the reassembly instructions for the valve body. I included all steps.
Additionally, it also shows the two check balls locations. I highlighted the heading on each of the relevant pages.
I hope this helps. Let me know how things turn out for you or if you have other questions,
Repair Safety Notice: This information is for general instructional purposes only. Vehicle repair can be dangerous. Verify all information, follow manufacturer service procedures, use proper tools and safety equipment, and consult a qualified repair shop when needed.