modified main class and Ubuntu class