Default Quin Jumping Animation

I added the Flying option that Matt Aspland did on YouTube and when I finished I lost my jump animation and its just a stiff jump.

I noticed in the tree its normally setup to go from Locomotion to the other locomotion to trigger the main states. In the video we went straight from the locomotion to the output pose so I lost all the items in main state.

How can I add my jump back?

Thank you