Implemented the return button on the game menu