Add extra methods to mock