add db.py function