Add class to plugin