more work on the ui