Working on week 2, problem 2