Update extract_class.py