Cleaned code, fixed game not restarting bug