refactor gameover and menu screens