Update log-likelihood to include weights. I should test this further though.