a 41TE IS a 604

You REALLY need to check the TCM for codes before you start tossing parts at it, There are a couple of other things that can cause that problem.

A code reader won`t do, find someone with a REAL scan tool and check it.

Rick