Updated test to output generated code