Update main.cc

function for initializing array of matrices is added!