Adding ability for mock verification