Added plt.show() statement to force plotting