Add some comment about StateT