Sure. I added it to the first if statement.