working on the ui