Implemented Langevin Monte Carlo as an alternative sampling strategy. Still not fixing the problems I'm having, though.