add solution for problem #2