This MR adds a getter for retrieving the transition map for a current state.
This will create a new commit in order to revert the existing changes.
