Create problem5a.py