Problem 31 working I think