Exchenged pylab to plt