Added functionality to UI