Add visualization support and test