added challenge6.py