Added BST code