Add visualization capabilities