Step 2: One-hot encoding for categorical variables