Improve computation time of the Imax function

Computation time of the Imax() function needs to be improved.The use of the double loop slows down the code strongly.