Merge pull request #7 from TaiSHiNet/master

Fixed a couple things in models.py