Update for Week 4 Assignment

Hoping this is what Andy is looking for.