Week 4 assigment