Add class function