You can find the link to the problem here

I find the tutorial implementation a bit confusing. I would really appreciate if someone can help me with the above problem. Thanks.