Add visualization methods