R.G.
My son loves bathtime too, and he used to scream and get mad at the time to come out for a couple of times!
What I dediced to do is that I make his bathtime 30 minutes before a bottle, so when it is time to come out of the bath, I have the bottle prepared, so he will take his bottle almost right away!