Updated a comment.

The comment explains how I'm going to calculate the gradients.