Continue naive, non-Pythonic method with all tests passing
A little more Python like
Yet more Pythonic