added visualization method