Update to part C of hw 3