updated generated code to include required models