Add visualization testing