diff --git a/README.md b/README.md
index 43b8e74..0471de0 100644
--- a/README.md
+++ b/README.md
@@ -30,10 +30,11 @@ Then you can have some fun with your robot intern. These prompts were tested in
 PDF includes:
 
 > The "seminarske" folder contains pdf journal submissions with their corresponding .txt version. The first few lines of each file contain the title. Write and execute a shell one-liner to extract the titles corresponding to the file names into your context.
-Read the "razpored.txt" file to map titles to categories.  
+Read the "razpored.txt" file to map titles to categories. 
+> Before the works from each category, add:
+> \worksection{Title of category}
 > Write the following latex command for each of the works, keeping their order from razpored.txt intact:  
-> \includework  
-> {Title of category}  
+> \includework 
 > {seminarske/name_of_the_file.pdf}  
 > {Full title of the work}  
 
diff --git a/edit_this/70-pdf-includes.tex b/edit_this/70-pdf-includes.tex
index 7fcaf33..20ea0f3 100644
--- a/edit_this/70-pdf-includes.tex
+++ b/edit_this/70-pdf-includes.tex
@@ -1,79 +1,70 @@
+% At the start of each category:
+%   \worksection{Title of the category}
 % For each work, use the following command to include it:
 %   \includework
-%   {Title of category}
 %   {seminarske/name_of_the_file.pdf}
 %   {Full title of the work}
 
-% Razno / Miscellaneous
+\worksection{Razno / Miscellaneous}
+
 \includework
-{Razno / Miscellaneous}
 {seminarske/DigFor-25_paper_2.pdf}
 {WARNE: Orodje za zbiranje dokazov o zalezovalski programski opremi}
 
 \includework
-{Razno / Miscellaneous}
 {seminarske/DigFor-25_paper_8.pdf}
 {How to detect cryptocurrency miners? By traffic forensics!}
 
 \includework
-{Razno / Miscellaneous}
 {seminarske/DigFor-25_paper_10.pdf}
 {Was the clock correct? Timestamps and time anchors across operating systems}
 
 \includework
-{Razno / Miscellaneous}
 {seminarske/DigFor-25_paper_14.pdf}
 {Enhancing speaker identification in criminal investigations through clusterization and rank-based scoring}
 
-% IoT in avtomobili / IoT and cars
+\worksection{IoT in avtomobili / IoT and cars}
+
 \includework
-{IoT in avtomobili / IoT and cars}
 {seminarske/DigFor-25_paper_6.pdf}
 {Oblačna forenzična analiza Amazon iRobot Roomba sesalca}
 
 \includework
-{IoT in avtomobili / IoT and cars}
 {seminarske/DigFor-25_paper_4.pdf}
 {Hit and run: Forensic vehicle event reconstruction through driver-based cloud data from Progressive's snapshot application}
 
 \includework
-{IoT in avtomobili / IoT and cars}
 {seminarske/DigFor-25_paper_11.pdf}
 {Grand Theft API: A Forensic Analysis Of Vehicle Cloud Data}
 
-% Datoteke in sistemi / Files and systems
+\worksection{Datoteke in sistemi / Files and systems}
+
 \includework
-{Datoteke in sistemi / Files and systems}
 {seminarske/DigFor-25_paper_1.pdf}
 {Applying digital stratigraphy to the problem of recycled storage media}
 
 \includework
-{Datoteke in sistemi / Files and systems}
 {seminarske/DigFor-25_paper_7.pdf}
 {Forenzični vidiki zloženih datotečnih sistemov}
 
 \includework
-{Datoteke in sistemi / Files and systems}
 {seminarske/DigFor-25_paper_12.pdf}
 {Forenzična analiza naprave Steam Deck: Predstavitev in razširitev članka Well Played, Suspect!}
 
 \includework
-{Datoteke in sistemi / Files and systems}
 {seminarske/DigFor-25_paper_13.pdf}
 {Beyond the 3DS: A Cross-Platform Study Inspired by Nintendo 3DS Residual Data Analysis}
 
-% Mobilna forenzika / Mobile forensics
+\worksection{Mobilna forenzika / Mobile forensics}
+
 \includework
-{Mobilna forenzika / Mobile forensics}
 {seminarske/DigFor-25_paper_5.pdf}
 {Analyzing Deterministic and Heuristic Approaches to JPEG Fragmentation Detection}
 
 \includework
-{Mobilna forenzika / Mobile forensics}
 {seminarske/DigFor-25_paper_3.pdf}
 {Natančnost geolokacijskih metapodatkov slik pametnih telefonov}
 
 \includework
-{Mobilna forenzika / Mobile forensics}
 {seminarske/DigFor-25_paper_9.pdf}
 {Mobile Device Forensics: A Targeted Overview}
diff --git a/main.tex b/main.tex
index baf1ea4..519e6c9 100644
--- a/main.tex
+++ b/main.tex
@@ -123,34 +123,22 @@ All rights reserved.}\\[0.5cm]
 %   INCLUDE ALL THE PDFs
 %----------------------------------------------------------------------------------------
 
-% This is a hack to make sure works in the same category don't get duplicated sections in the ToC
+\newcommand{\includework}[2]{%
+    \includepdf[
+        pages=-,
+        pagecommand={},
+        addtotoc={1,subsection,1,{#2},{#2}}
+    ]{#1}%
+}
 
-% Store the last category/section that was used
-\def\lastsection{}
+\newcommand{\worksection}[1]{%
+    \newpage
+    \center % Center everything on the page
+    \vspace*{60pt}
 
-\newcommand{\includework}[3]{%
-    % Expand \lastsection before comparison
-    \edef\tmplastsection{\lastsection}%
-    \edef\tmpcurrentsection{#1}%
-    \ifx\tmpcurrentsection\tmplastsection
-        % Same as last → only add subsection
-        \includepdf[
-            pages=-,
-            pagecommand={\thispagestyle{plain}},
-            addtotoc={1,subsection,1,{#3},{#3}}
-        ]{#2}%
-    \else
-        % Different → add section and subsection
-        \def\lastsection{#1}% Update the last section
-        \includepdf[
-            pages=-,
-            pagecommand={\thispagestyle{plain}},
-            addtotoc={
-                1,section,1,{#1},{#1},
-                1,subsection,1,{#3},{#3}
-            }
-        ]{#2}%
-    \fi
+    \section{{#1}}
+
+    \vfill % Fill the rest of the page with whitespace
 }
 
 % Include all the PDFs