Update for B5 Assignment