Made small changes to UI and code structure for lobby.