added challenge4.py