using matplotlib.pyplot instead of pylab