python solution for problem 9