import matplotlib first