Add an implementation of the new method to the mock