Added game over and restart