Replaced plt.draw() with plt.show() and remove transpose