modify sample.py file