Adding the ability to mock scalar UDFs