New idea of implementation - retrieve the UnnormalisedDensity from the state space.