New class for UI (add .java)