Return transform stream instead of function