Create ex_3_1.py