Added test for max, it seems to work in 2D