playing around with the python implementation