Добавлена функция plt.savefig для вывода графиков