Updating in-class assignment Week04