Organized UI code in lobby