More lecture code
Week 6 assignment