Add basic support for snapshots