plot method added for test_count