Update models.py

Fixed wrong import in models.py
Merge pull request #5 from ewokcillo/master

Fixed wrong import in models.py