python solution for problem 31