completes Week 5 + Problem Set