Added an edit and delete function