is it possible to code a semi-aquatic mob like the axolotl?

[round snow]

I want my alligator to have axolotl movement

[sand pelican]

why not see the code for axolotl and then adding it to your mob?