Add support for using Grain snapshots