Create ex_2_1.py