Added visualization patch

Changed original visualization patch to include a subpatch that handles
all the visualization code. Cleaned up the edit view.