Amended if statement
Added comment