minors changes to the UI