Made some changes + comments in intro (Figure 1 and notation)