Added missing comment class. Woops.