Adding classes for graph