Code for codility equilibruim