Added lobby functionality