problem4 done