I made some changes.

I added a vector and changed the plot variables.