solved problem 8 in python