fixed big bug and added restart_game