add test for plot