Fixed problem with activation function.