Added plots and modifeid the code

plots added