Add code for enabling and disabling streams