Added support for explicit belief state updates