added week3 code