Create week1_problem1.py