Made changes to UI