Minor bug in likelihood calc - fixed with updated test cases