- add plotting