Updated Gaussian output layer