diff --git a/buggy.py b/buggy.py
--- a/buggy.py
+++ b/buggy.py
@@ -7,6 +7,7 @@
     return res
 
 
+MOD = 998244353
 N = int(input())
 A = list(map(int, input().split()))
 S = list(reversed(prefix_sum(list(reversed(A)))))
@@ -17,4 +18,5 @@
 res = 0
 for i in range(N - 1):
     res += A[i] * ditsum[i + 1] + S[i + 1]
+    res %= MOD
 print(res)
