moved code around
fixed the game restart