Added import on models, plus fixed typo in line 104/5
Added import on models, plus fixed typo in line 104/5
Merge pull request #7 from TaiSHiNet/master

Fixed a couple things in models.py