Create challenge4_1.py