Adjusted structure to clarify.
Created template for softmax.