update assignment 2

update assignment 2