Create problem5b.py