Updated comment in last code block