add class to models.py