softmax layer
some minor fixes
fix