Add second part of Week4 Ex1. Needed help on this, but eventually I understood.