private List<TransitLPActivityUnit> getActivityUnits() {
    ArrayList<TransitLPActivityUnit> activityUnit = new ArrayList<TransitLPActivityUnit>();
    System.out.println("Size : " + view.getUnitChoices().size());

    for (List<Record> unitRecord : view.getUnitChoices()) {
      TransitLPActivityUnit unit = new TransitLPActivityUnit();
      unit.setCategory(transitCategory.getId());

      for (Record record : unitRecord) {
        System.out.println(
            "CHOICe activity to be added : "
                + ((TransitLPActivity) record.getAttributeAsObject("transit")).getName());
        unit.addActivity(((TransitLPActivity) record.getAttributeAsObject("transit")).getId());
      }

      activityUnit.add(unit);
    }

    return activityUnit;
  }