Added support for mocking functions inside of modules