exercise4.py