Cleanup the generated code