small changes to the UI