Hello,
Were you able to figure out which part to place the code?
I’m stuck here as well, having the same issue.