final changes to the ui