problem 5 & 6 done