updated code generation too to use RazorMachine