made a few more alterations to the UI