updated hw2 prob3