adding visulization stuff
adding visulization stuff