Update db_wrapper.py