add plotting graphs and code