add solution to problem #2