added functionality to UI