Still working on extract.py