Minor changes

TODO: Make a MenuScreen that GameOver inherites from