Commented where I think the problem is