Adding function stream_take