Tree Traversal Python Code