Wrote forward-backward sampler and added current number of states as a field