modify homework2