updates for edit functionality