Add game over screen, better collisions