Update retrieve_db.py