Tweaks to small flow example