first version for Models.BayesianWorkingMemory