added test plotting