Add lobby (not finished yet) and move the main method