issues
started wrking on visuals
fixing minor issues with visualization