Add UserInterface as main class