Added classes for correlation graph