adds plotting in benchmark