Add checkpointing for DStreams

Fixes #64