minor changes to the UI