Small tweaks to the ui