@etrain Could you also update this line? We should only touch the diagonals.