Update to lastest python code