avl_tree: Cleanup and improve comments