use db.py to connect