Add 'edit' functionality