working on the ui in class