added tests for plot