Updated game over functionality