Add main and run_game functions