Adding BST code