Add diag() function