Extend Python abstract syntax tree
