Improving readme. Pass argument for plotting script.
Remove unnecessary packages from plotting script.