Just a small PR to allow setting top level comment on a state machine By submitting this pull request, I confirm that my contribution is made under the terms of the Apache-2.0 license