Add support for snapshot module