3 files changed, 11 insertions, 10 deletions
diff --git a/src/database/sql/fetch_display_submissions.sql b/src/database/sql/fetch_display_submissions.sql
index 5e836b3..aaceaca 100644
--- a/src/database/sql/fetch_display_submissions.sql
+++ b/src/database/sql/fetch_display_submissions.sql
@@ -3,8 +3,9 @@ submission.id AS id,
 user.username AS username,
 submission.language AS language,
 submission.details AS details,
-submission.code AS code
+submission.code AS code,
+submission.code_length as code_length
 FROM submission JOIN user
 ON user.id = submission.user_id
 WHERE submission.problem_id = ?1
-ORDER BY length(submission.code);
+ORDER BY submission.code_length;
diff --git a/src/database/sql/initialize.sql b/src/database/sql/initialize.sql
index 8c0f2ef..3569fd5 100644
--- a/src/database/sql/initialize.sql
+++ b/src/database/sql/initialize.sql
@@ -13,13 +13,13 @@ CREATE TABLE IF NOT EXISTS user (
 );
 
 CREATE TABLE IF NOT EXISTS submission (
-    id         INTEGER PRIMARY KEY,
-    user_id    INTEGER NOT NULL,
-    problem_id INTEGER NOT NULL,
-    language   TEXT NOT NULL,
-    details    TEXT NOT NULL,
-    code       TEXT NOT NULL
-    
+    id          INTEGER PRIMARY KEY,
+    user_id     INTEGER NOT NULL,
+    problem_id  INTEGER NOT NULL,
+    language    TEXT NOT NULL,
+    details     TEXT NOT NULL,
+    code        TEXT NOT NULL,
+    code_length INTEGER NOT NULL
 );
 
 INSERT INTO problem (title, description) VALUES (
diff --git a/src/database/sql/insert_submission.sql b/src/database/sql/insert_submission.sql
index 96df322..3f70bca 100644
--- a/src/database/sql/insert_submission.sql
+++ b/src/database/sql/insert_submission.sql
@@ -1 +1 @@
-INSERT INTO submission (user_id, problem_id, language, details, code) VALUES (?1, ?2, ?3, ?4, ?5) RETURNING id;
+INSERT INTO submission (user_id, problem_id, language, details, code, code_length) VALUES (?1, ?2, ?3, ?4, ?5, ?6) RETURNING id;