Changes -- fixed visualization, some convenience changes and some bug fixes.