Continued working on AVL tree