Add edit and delete functions