working on the UI