Start of fixing log likelihood