added matplotlib import