Adds support for streaming, minor fixes, wip