Add class to h2