zero_diag and diag finished, tests written for mean and sum