Add first version of a python implementation