Small adjustments to the ui