New code

I've marked the line in question with a comment before it