fixed pyplot import