fix visualization plot3d
added visualization 2d