Changed pylab dependency to matplotlib.pyplot dependency