Exercism series in Python