Added visualizer code
Small fixes
Updated readme