corrected momentum sum array