Update models in import
review fix
flake correction
Merge pull request #266 from kubked/fix/categories-import

Update models in import