Better axolotl taming

Hi, I was playing around with axolotls, and I noticed that unless you have a lead, having a axolotl follow you around is quite difficult. I propose when an axolotl is tamed (picked up in a bucket), the axolotl will try to follow the player when they are in water (if the player is on land then the axolotl should stay in water, and if the axolotl is on land, then they should prioritise finding water). Also, I propose that when an axolotl is killed (which is something that could happen if they are fighting drowned/guardians) then the player who bucketed them should get a death message as the player does when their wolf/cat/parrot die.

EDIT: A sitting behaviour should also probably be added in case the player wants their axolotl out of their bucket but doesn't want the axolotl to follow them.