added visualization capabilities