assign2 update

assign2 udpate