Small changes to binary tree