update CBB520HW2_Part3 script