add plotting function