Update ex10_1.py

Added two lines (2, 4) from 10_2