First cut at week2 assignment