Clean up
Cleaner interface for the mock objcet.