Minor change: added some brackets to clarify if statement.