Update to Assinment 3