java edition trident need upd

I want Impaling work when the mob is in water/rain like bedrock in java edition and other is when you throw the trident go back in same spot I mean for example I have my trident number 7 of hotbar when I throw in java when is back is put 1 so I think cool Is 7 back to 7 or I think cool idea is you have in second hand back you throw you trident is back to second hand this no happen 1.21.9 when I have second hand I throw the trident back to hotbar no second hand (sorry for my bad english I try my best) I think this make more fun trident more useful because now only trident work is for 2 option channeling for creeper or building and riptide to move fast with or without elytra I hope see this and make trident better :D