removed unnecessary import of matplotlib