Create extract_class.py