Improved lobby UI, added missing java files