Add some missing imports to the examples in the models documentation.