Finished #26 in python