Added better syntax for controlling mocks