Added main class for GUI