np.max -> np.amax