Solution to python-koans