python solution for problem 11