Simplified the 2d equation visualiser a bit