add solution (python) for problem 22