Update sub_sample.py