Adding variable to check that weights for matrices are return correctly