Adds gameover funcitionality
Fixes display