Fixing matplotlib import error