Add class for feature computation