modified:   balanced_binary_tree/solution.py