added more to db.py