Done softmax output layer and  ReLU activation function. Model included.