Add stream limiting