Update models.py

added Training class.