added models/db.py