python solution for problem 8