Added snapshot testing code