Building the baseline model