Added log likelihood

Plus more testing