Added basic function mocking