Baby bath

We just moved into a new house but it only has walk in showers☹️ I have a 5 month old and I’m expecting another baby.. right now he has a baby bathtub that I bathe him in but bath time was my favorite as a kid and I want them to be able to have a fun bath as well lol, is there something I could buy that he could fit in as he gets older to be able to take a bath and play?

this is him in his bathrooms shower