Add edit and delete functionality