Lol, it was working fine on the basis of some posterior prediction. But then I saw this
Just looks like it’s not going to be simple to make this model work