added visualization, some more changes