Create problemset2a.py