add better visualization