hw4/: prob 2 done