Add diagonals back in.