Week 6 - Assignment  5 - Partial solution