Added Week 6 Code