What may be easier is to move the heating element to the top, and add a pump to move water to the top when heating.

Hot water will stay at the top of the tank, so that means you can use most of the hot water without a significant drop in temperature.

There's a company making tanks like this: https://www.youtube.com/watch?v=z1Z4JCoPAGc