Made changes to the UI