Week 4 assignment continued...