Added code to set lobby