Fixing trailing whitespace, line length.
Renamed module.
Adding matplotlib module for testing.