added UI level functionality of the update / remove buttons