Add tests for updating and deleting agents