Features engineering for the baseline model