Added a class to find the 2D peak