Fixing edit/delete/save button functionality