Selecting a checkbox in selenium

selectCheckbox.java

private final WebDriver driver;

protected void setCheckbox(By locator, boolean desiredState) {
      WebElement element = driver.findElement(locator);
      boolean selected = element.isSelected();
      if (checkboxNeedsToBeClicked(selected, desiredState)) {
          element.click();
      }
}

private boolean checkboxNeedsToBeClicked(boolean selected, boolean desiredState) {
      return !(selected == desiredState);
}