Small changes to the UI