more work up for the ui