modify homework 2