Week 2. Problem 4 complete