Working on end game class