Added end-game class