#yoloswag: added models.py