added linear activations and mean square loss