added a simple plot at the end.

minor edits done.