Update for latest assignment