Create week2_problem2.py