Better Python AST node code generation