Added functionality to return css classes