add database model to models.py