Finished Week 6 assignment