Make flying pets disperse when die + other ideas

The idea is that when you die currently the flying pets freeze in place when you die


like this

what i think would be cool is having those pets fly outwards


like this, or another idea is make them fly upwards, or even just make them shrink and disappear
then when you respawn they come back
i feel something like this would be cool