Feature engineering for the baseline model