Add class for main block