Tried to add a better method for detecting JumpStage

Untested