Extend package for classification and regression