This is the week 2 assignment