Just added some classes for detecting faults with a deep belief network