Week 3 in class code updated