Add stop stream functionality