Balanced Binary Tree in Python