StateFn godoc tweak
make the example a little more complete
fix doc comments