add plot function