var grid = [

    [false, false, false, false, false],

    [false, false, false, false, true],

    [false, false, false, true, true],

    [false, false, false, false, true],

    [false, false, false, false, false]

  ];

​

function setup() {

  createCanvas(400, 400);

  stroke(255, 0, 0);

  // for(var i = 0; i < 5; i++){

  //   for(var j = 0; j < 5; j++){

  //     grid[i][j] = boolean(int(random(0,2)));

  //   }

  // }

}

function draw() {

  var won = true;

  for (var i = 0; i < 5; i++) {

    for (var j = 0; j < 5; j++) {

      if (grid[i][j] == true) {

        won = false;

      }

    }

  }

  if (won == true) {

    stroke(0);

    background(0);

    fill(255);

    textAlign(CENTER);

    textSize(24);

    text('Congratulations, you won! \n Click anywhere to try a new puzzle!', 200, 200);

    if(mouseClicked() == true){

      stroke(255, 0, 0);

    }

  } else {

    drawGrid();

  }

}

function mousePressed() {

  let col = int(mouseX / 80);

  let row = int(mouseY / 80);

  flipLight(row, col);

  flipLight(row - 1, col);

  flipLight(row + 1, col);

  flipLight(row, col - 1);

  flipLight(row, col + 1);

}

function flipLight(r, c) {

  if (isValid(r, c) == true) {

    grid[r][c] = !grid[r][c];

  }

}

function isValid(r, c) {

  if (r >= 0 && r < 5 && c >= 0 && c < 5) {

    return true;

  } else {

    return false;

  }

}

function drawGrid() {

  for (var r = 0; r < grid.length; r++)

    for (var c = 0; c < grid[r].length; c++) {

      let x = c * 80;

      let y = r * 80;

      if (grid[r][c] === false) fill(0);

      else fill(255);

      rect(x, y, 80, 80);

    }

}

​