Add snapshot handling