Started implementing visualizer and a little better code