Pulling up success flag functionality into Maestro base.

@stephanh Can you please review flag creation , we can move this up here.
Is the yield part actually needed?