expansion of states for two arms now possibly works, not tested.