Added visualization of 2d