update to latest python code