Updated the sample.

The sample now supports the unit to transform from states to
states/sub-states.