Minor Update

Cleaning up the UI