A code Kata in Python