added visualization stuff
viz stuff works
updated readme