Updates Week 3 code