Update models.py

change # of firms