need a little more engine, ignition, carb, etc. info to make any suggestions here..
and do yourself a favor, pick one thing at a time to adjust / replace or you'll end up in circles..
and agree with above, trans shop should have some idea what they may have done to affect those symptoms..
they must have had some comments regarding drivability during their test drive..
based on these circumstances, I was thinking;
- cracked distributor cap or cocked out of position
- any plug wires pulled may have been crossed, check 18436572
- distributor hold down may be loose or have loosened in the shop
- vacuum modulator hose damaged or some other large vacuum leak
posting some basic specs, what you know or can prove by testing, and what you've done would help..
good luck!