Tree Coding Python