Refactored endscreen to be a menu