Python solution to problem 6.