some small katas in python