python solution for problem 12