added the 'edit' feature