fixed SoftMaxTree.update issue. Still looking for NaN bug...