Add code to handle gameover and start game again