[WIP] Add avl tests

Adding AVL tests here.