class Matrix {
  float[][] M;
  int  m, n, i, j;
  
  Matrix(int _m) {
    m=_m;
    n=_m;
    M = new float [m][n];
  }

  Matrix(int _m, int _n) {
    m=_m;
    n=_n;
    M = new float [m][n];
  }

  void Set(float [] data) {
    int k=0;
    for(i=0; i<m; i++)
      for(j=0; j<n; j++)
        M[i][j]=data[k++];
  }

  void Identity() {
    if(m==n) {
      for(i=0; i<n; i++) 
        M[i][i]=1;
    }    
  }  
  
  void Add(Matrix a, Matrix b)
  {
    for(i=0; i<m; i++) {
      for(j=0; j<n; j++) 
        M[i][j]=a.M[i][j]+b.M[i][j];
    }          
  }    
  
  void Print() {
    for(i=0; i<m; i++) {
      for(j=0; j<n; j++) 
        print(M[i][j]+" ");
      println(" ");  
    }    
  }
}