better arrangement for tracking learning trajectories, finished Rescorla-Wagner model