Edit db.py to connect correct database and add data to database