2d plots are also working