Week 3 assignment- with comments