Added restart functionality on spacebar press