Add extract_test.py