Complete modifying the Main.m, Next: OOA and plot