Changed from class feature extraction to regression setup