updated week 4 code with a comment