problem 3 done
problem 4 done