Fix styling issues with add timeout to stream with testing