Refactor load_menu_game/main function some more