[experiment] better visualization + bug fixes.