Added visualization features