Minor changes to the UI's