MJ baths in a Tummy tub and is always content afterwards, sometimes after a shower or a big baths she gets really cranky and screams... we find that if we get her out the moment she looks tired or over it, it helps.
The other thing to try is getting a wheat bag and having that on the towel so that it is toasty and warm when she is put on it. (i also use the heat pack in the cot before she goes down at night and if i have to give her a night feed (I love it afterwards )
I now use the TT in the lounge and the towel next to it, as soon as she is lifted out it is straight onto the towel and wrapped up and we play gentle games (like blowing on her face lightly, or raspberrys on her tum or just laughing) for a while before the massage and getting dressed which can make her cold and more likely to have a cry.

Hope this helps