Is there a way to make Water Creatures swim Up and Down?

Is there currently any way to make Water Mobs Have the Ability to swim Up and Down? They only seem to be able to go along a Flat Plane, and it would not be very terrifying if a kraken was stuck on the surface of the water, which makes it easy to spot.