matplotlib isn't used either.

PEP8: there's a line missing between the import and the class (there should be two blank lines)