finished tree indexing in SoftmaxTree constructor
Using nn.Modules for SoftmaxTree:forward
finished SoftmaxTree:forward. Will require its own TreeNLL Loss.