assignment02 updated