Create retrieve_db.py