Week 2 assignment update