BST testing code modified