Fix peak_local_max doc string
Add test case for 3D images
Add test case for 4D images