Simple matplotlib transmission line simulation