Create challenge6_1.py