remove import of matplotlib