Add avl tree balancing