add better edge visualization