Tiny modification
Add AVL tree