Add collect-stream

Sorry I missed this one in my earlier PR