Update extract.py