Added week 1 code