Allay Upgrades

Allays will no longer forget their note block when they are too far away for too long (usually distracted by items) instead they will only forget their note block when they are close enough to hear it and tell that it's not playing.

Placing an amethyst block under a note block will cause the allays to never* forget that noteblock, meaning you can make an allays stay in an area without the constant ringing of note blocks.
You can make them forget a noteblock by punching them.

If an allays is too far away from their owner and not following them anymore, they will teleport to them. (like how a wolf will teleport).

Untamed allays will no longer fly too high into the sky to be given an item.

Store untamed allays by pushing them above an amethyst block. They will have amethyst particles and will not move from that spot until you give them an item.

Allays with items will automatically dismount other entities, so they won't get stuck in boats or minecarts.

Allay can be stored in books for easy transportation or to reduce lag. Drop a book on an allay to store it in that book. Drop an allay book on the ground to release the allay.