adding support for mock testing