Added test plotting module