Fix a minor bug - Added a few models