Added test plotting routine