Change function plot()