updates to prototype flow