add plotting