Add if statement.