add class to the main container