add mock function