updated sample flow, needs fix