Click on the stage to end but stay watch press space to go back.
Credits: @aneeshsnpdragon for helping me code this(my brother).
