public void draw() {
    background(c.black);

    if (!menu.startPressed()) {
      menu.display();
    } else {
      if (allowChoice) {
        if (menu.difficulty == 0) {
          // Base :)
        } else if (menu.difficulty == 1) {
          levels.level1.complete = true;
        } else if (menu.difficulty == 2) {
          levels.level1.complete = true;
          levels.level2.complete = true;
        }

        allowChoice = false;
      }
      levels.display();
      if (levels.isQuitPressed()) {
        resetAllData();
      }
    }
  }