Some changes made to UI