Solved problem 8 in Python