Added exercise4.1.py