For week 2 assignment