Octane "Heat" Mechanic

Octane losing health from the massive speed boost he gets is the balancing side of his ability as movement in the game is such a crucial part of it. Octane does need a buff. but I don't know if changing his stim is where the change needs to come from. Perhaps when using his ultimate when you double jump you can change direction in the air. or while he's using stim he can run on walls much like the wall run mechanic in Titanfall.