Week 2 assignment work