Generated some graphs and cleaning up the code for presentation.
removing extra imports.