Edit functionality implemented in UI