Create problem2b.py