Added some documentation for softmax and cross entropy