Sorry for your problems' I'll try to help.
The battery need not be disconnected when replacing the CPS.
They should have reset the P0335 code, if it is still there, then either they didn't, or it has come back.
Go back and have them reset the code, then wait to see if it comes back.