Add class for preview