adding tests for plotting