(crappy) solution to problem 5 added