fixed visualize, added 2 columns, many other fixes