Week 6 assignment continued...