So preparing your data with the correct column to help model can be a specific step and store in another variable. pipe with . works but you mix steps.
You should give a look at
and specifically the part about Model. It could help you with good practice.
Also, if your question's been answered (even by you!), would you mind choosing a solution? It helps other people see which questions still need help, or find solutions if they have similar problems. Here’s how to do it: