update step4 solution