Added Visualization

TODO: Make it look prettier