made a few changes to the UI