Added week 4 code