Assignment 3b complete