some work on the UI