Adding better visualization method