Create Part b 2.py