Added code to lobby