fixing likelihood to return vector