Updating the BST class with some comments