work on the UI