added week 4 code