changed 2d plot to 3d

still haven't actually tested it