Cleaned up code, started implementing a game restart button