Added functionality to load menu after the game ends