Add back the AVL Tree implementation