Fixed bug with restart

Added event listener to listen for game restart