Problem 50 not quite working