I applied more gravity for every second you are in the air as the mechanic. But a stamina bar can work too if you implement it well!