Add diameter to bst