Add the delete function on unit test