fixed matplotlib import