Added Week 4 base code