commented out pyplot.show() in main declaration