touching up the UI