Added pylab import. Anyway, it is needed, if we want to use shortcuts to functions.