Really adding tree traversal code :)