need to fix plotting of points