Update avl_tree.py