Finished #34 in python