Remove unused files. Modify main to plot graph