Also introduce check for mass conservation in prob_laplace. Currently does not sum to 1, a factor of 2 is missing somewhere.