Added logic for game over and restart game