Hi, you need to indent your code correctly. I suggest you to spend a few minutes to read this post, and make changes to your code for indentation. Also, you have 3 code lines starting with z_wb, and I suppose the first one is not intended to be there and please remove it.
Raymond
PS: we can’t share assignment code here, so I am removing it.