Balanced Binary Tree: cleaner solution