Add class param