For week 1 assignment