better comments and variable matrix sizes