add solution to section 4