Week1, assignment 2, UNQ_C4 , define model function

I figured it out. It was related to Y and the solution was in this post: