Update mock_db.py