diff --git a/notes.org b/notes.org
index 1a39681..2fa35c6 100644
--- a/notes.org
+++ b/notes.org
@@ -2,10 +2,6 @@
 * Tests
 ** Look for problem cases in other latex documents
 * Features
-** Fold long lines to 80 characters
-*** Care with trailing comments with space or no space
-*** No folding in verbatim environments, just warn
-** Merge short alphanumeric lines
 ** Flags
 *** -d Dry run
 *** -p Print to STDOUT
diff --git a/tests/environment_lines_in.tex b/tests/environment_lines_in.tex
index 80184ce..55a0b71 100644
--- a/tests/environment_lines_in.tex
+++ b/tests/environment_lines_in.tex
@@ -1,3 +1,11 @@
+\documentclass{article}
+
+\begin{document}
+
+\newenvironment{env1}{}{}
+\newenvironment{env2}{}{}
+\newenvironment{env3}{}{}
+
 % environments on separate lines
 \begin{env1}
 \begin{env2}
@@ -16,4 +24,6 @@
 \begin{env1}\begin{env2}\end{env2}\end{env1}
 
 % environments with extra brackets
-\begin{env1}()(\begin{env2}[{}]\end{env2}[])\end{env1}
+\begin{env1}(a)(b \begin{env2}[c{d}e] \end{env2}[f]g)\end{env1}
+
+\end{document}
diff --git a/tests/environment_lines_out.tex b/tests/environment_lines_out.tex
index 6df88ff..1d4b42d 100644
--- a/tests/environment_lines_out.tex
+++ b/tests/environment_lines_out.tex
@@ -1,3 +1,11 @@
+\documentclass{article}
+
+\begin{document}
+
+\newenvironment{env1}{}{}
+\newenvironment{env2}{}{}
+\newenvironment{env3}{}{}
+
 % environments on separate lines
 \begin{env1}
   \begin{env2}
@@ -23,7 +31,9 @@
 \end{env1}
 
 % environments with extra brackets
-\begin{env1}()(
-    \begin{env2}[{}]
-  \end{env2}[])
+\begin{env1}(a)(b
+    \begin{env2}[c{d}e]
+  \end{env2}[f]g)
 \end{env1}
+
+\end{document}
diff --git a/tests/lists_in.tex b/tests/lists_in.tex
index 488c724..ae692ca 100644
--- a/tests/lists_in.tex
+++ b/tests/lists_in.tex
@@ -1,3 +1,7 @@
+\documentclass{article}
+
+\begin{document}
+
 \begin{itemize}
 
 \item Lists with items on one line
@@ -13,3 +17,5 @@
 Or even just % trailing comments
 
 \end{itemize}
+
+\end{document}
diff --git a/tests/lists_out.tex b/tests/lists_out.tex
index d7b51a8..9d933fc 100644
--- a/tests/lists_out.tex
+++ b/tests/lists_out.tex
@@ -1,3 +1,7 @@
+\documentclass{article}
+
+\begin{document}
+
 \begin{itemize}
 
   \item Lists with items on one line
@@ -13,3 +17,5 @@
     Or even just % trailing comments
 
 \end{itemize}
+
+\end{document}
diff --git a/tests/masters_dissertation_in.tex b/tests/masters_dissertation_in.tex
index da54e7d..7468896 100644
--- a/tests/masters_dissertation_in.tex
+++ b/tests/masters_dissertation_in.tex
@@ -1,4 +1,4 @@
-\documentclass[12pt]{ociamthesis}
+\documentclass[12pt,draft]{ociamthesis}
 
 %TC:ignore
 
@@ -45,15 +45,15 @@
 \usepackage{subcaption}
 
 % draft options
-\usepackage{ifdraft}
-\ifoptiondraft{
-\usepackage{draftwatermark}
-\SetWatermarkText{DRAFT}
-\SetWatermarkScale{6}
-\SetWatermarkColor[rgb]{1,0.9,0.9}
-\usepackage{showframe}
-\usepackage{layout}
-}{}
+%\usepackage{ifdraft}
+%\ifoptiondraft{
+%\usepackage{draftwatermark}
+%\SetWatermarkText{DRAFT}
+%\SetWatermarkScale{6}
+%\SetWatermarkColor[rgb]{1,0.9,0.9}
+%\usepackage{showframe}
+%\usepackage{layout}
+%}{}
 
 % hyperlinks
 \usepackage[plainpages=false,draft=false
@@ -215,7 +215,7 @@
 \begingroup
 \let\cleardoublepage\relax
 \let\clearpage\relax
-\printglossary[title=Abbreviations, style=mystyle]
+%\printglossary[title=Abbreviations, style=mystyle]
 \endgroup
 
 \end{romanpages}
@@ -404,8 +404,8 @@ \section{Graph definitions} \label{sec:graphs_graph_definitions}
 %
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.7,draft=false]{%
-../tikz/motif_definitions_directed/motif_definitions_directed.pdf}
+%\includegraphics[scale=0.7,draft=false]{%
+%../tikz/motif_definitions_directed/motif_definitions_directed.pdf}
 \caption{All simple motifs on at most three vertices}
 \label{fig:motif_definitions_directed}
 \end{figure}
@@ -810,16 +810,16 @@ \subsection{Eigenvector sweep} \label{sec:spectral_sweep}
 \begin{figure}[H]
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../tikz/eigenvector_sweep_network/eigenvector_sweep_network.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../tikz/eigenvector_sweep_network/eigenvector_sweep_network.pdf}
 \caption{A small network}
 \label{fig:eigenvector_sweep_network}
 \end{subfigure}
 %
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/eigenvector_sweep/eigenvector_sweep_scores.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/eigenvector_sweep/eigenvector_sweep_scores.pdf}
 \caption{Sweep profile of the network}
 \label{fig:eigenvector_sweep_profile}
 \end{subfigure}
@@ -1001,8 +1001,8 @@ \subsection{Symmetric two-block DSBMs}
 
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.8,draft=false]{%
-../tikz/sym_two_block_dsbm/sym_two_block_dsbm.pdf}
+%\includegraphics[scale=0.8,draft=false]{%
+%../tikz/sym_two_block_dsbm/sym_two_block_dsbm.pdf}
 \caption{Symmetric two-block DSBM block structure and sparsity matrix}
 \label{fig:sym_two_block_dsbm}
 \end{figure}
@@ -1024,14 +1024,14 @@ \subsection{Symmetric two-block DSBMs}
 \begin{figure}[H]
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/motifsym/motifsym_1.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/motifsym/motifsym_1.pdf}
 \caption{$n=50$, $p=0.3$, $q=0.2$}
 \end{subfigure}
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/motifsym/motifsym_2.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/motifsym/motifsym_2.pdf}
 \caption{$n=100$, $p=0.15$, $q=0.1$}
 \end{subfigure}
 \caption{ARI violin plots for the symmetric two-block DSBM}
@@ -1051,8 +1051,8 @@ \subsection{Asymmetric two-block DSBMs} \label{sec:motif_asymm_dsbms}
 
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.8,draft=false]{%
-../tikz/asym_two_block_dsbm/asym_two_block_dsbm.pdf}
+%\includegraphics[scale=0.8,draft=false]{%
+%../tikz/asym_two_block_dsbm/asym_two_block_dsbm.pdf}
 \caption{Asymmetric two-block DSBM block structure and sparsity matrix}
 \label{fig:asym_two_block_dsbm}
 \end{figure}
@@ -1075,14 +1075,14 @@ \subsection{Asymmetric two-block DSBMs} \label{sec:motif_asymm_dsbms}
 \begin{figure}[H]
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/motifasym/motifasym_1.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/motifasym/motifasym_1.pdf}
 \caption{$n=100$, $p=0.2$, $q_1=0.35$, $q_2=0.05$}
 \end{subfigure}
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/motifasym/motifasym_2.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/motifasym/motifasym_2.pdf}
 \caption{$n=200$, $p=0.15$, $q_1=0.25$, $q_2=0.05$}
 \end{subfigure}
 \caption{ARI violin plots for the asymmetric two-block DSBM}
@@ -1115,15 +1115,15 @@ \section{US Political Blogs network} \label{sec:motif_polblogs}
 \begin{figure}[H]
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/polblogs/polblogs_network.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/polblogs/polblogs_network.pdf}
 \caption{The US Political Blogs network}
 \label{fig:polblogs_network}
 \end{subfigure}
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/polblogs/polblogs_ari_conn.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/polblogs/polblogs_ari_conn.pdf}
 \caption{ARI against $|C|$ across motifs}
 \label{fig:polblogs_ariplot}
 \end{subfigure}
@@ -1143,15 +1143,15 @@ \section{US Political Blogs network} \label{sec:motif_polblogs}
 \begin{figure}[H]
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/polblogs/polblogs_M12_truth.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/polblogs/polblogs_M12_truth.pdf}
 \caption{Colouring by truth label}
 \label{fig:polblogs_embedding_truth}
 \end{subfigure}
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/polblogs/polblogs_M12_clusts.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/polblogs/polblogs_M12_clusts.pdf}
 \caption{Colouring by $k$-means++ cluster}
 \label{fig:polblogs_embedding_kmeans}
 \end{subfigure}
@@ -1192,20 +1192,20 @@ \section{US Migration network} \label{sec:motif_migration}
 \begin{figure}[H]
 \begin{subfigure}{.325\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/us_migration/us_migration_sweep_profile_Ms.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/us_migration/us_migration_sweep_profile_Ms.pdf}
 \caption{$\ca{M}_\mathrm{s}$}
 \end{subfigure}
 \begin{subfigure}{.325\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/us_migration/us_migration_sweep_profile_M6.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/us_migration/us_migration_sweep_profile_M6.pdf}
 \caption{$\ca{M}_6$}
 \end{subfigure}
 \begin{subfigure}{.325\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/us_migration/us_migration_sweep_profile_M9.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/us_migration/us_migration_sweep_profile_M9.pdf}
 \caption{$\ca{M}_9$}
 \end{subfigure}
 \caption{Sweep profiles of the US Migration network}
@@ -1257,7 +1257,7 @@ \section{US Migration network} \label{sec:motif_migration}
 \centering
 \setlength{\tabcolsep}{0em}
 \begin{tabular}{ |c|c|c|c| }
-\expandableinput ../../results/us_migration/us_migration_table.txt
+%\expandableinput ../../results/us_migration/us_migration_table.txt
 \end{tabular}
 \end{table}
 \vspace*{-0.5cm}
@@ -1299,7 +1299,7 @@ \subsection{Collider and expander motifs} \label{sec:coll_expa}
 
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.8,draft=false]{../tikz/expa_coll/expa_coll.pdf}
+%\includegraphics[scale=0.8,draft=false]{../tikz/expa_coll/expa_coll.pdf}
 \caption{The collider and expander motifs}
 \label{fig:expa_coll}
 \end{figure}
@@ -1396,8 +1396,8 @@ \section{Bipartite stochastic block models} \label{sec:bipartite_sbms}
 
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.8,draft=false]{%
-../tikz/bipartite_dsbm/bipartite_dsbm.pdf}
+%\includegraphics[scale=0.8,draft=false]{%
+%../tikz/bipartite_dsbm/bipartite_dsbm.pdf}
 \caption{BSBM block structure and sparsity matrix}
 \label{fig:bipartite_bsbm}
 \end{figure}
@@ -1418,14 +1418,14 @@ \section{Bipartite stochastic block models} \label{sec:bipartite_sbms}
 \begin{figure}[H]
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/bipartite/bipartite1.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/bipartite/bipartite1.pdf}
 \caption{$n=100$, $p=0.2$, $q=0.1$}
 \end{subfigure}
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/bipartite/bipartite2.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/bipartite/bipartite2.pdf}
 \caption{$n=200$, $p=0.1$, $q=0.06$}
 \end{subfigure}
 \caption{ARI violin plots for the BSBM}
@@ -1452,15 +1452,15 @@ \section{American Revolution network} \label{sec:bipartite_american_revolution}
 \begin{figure}[H]
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/american_revolution/american_revolution_source.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/american_revolution/american_revolution_source.pdf}
 \caption{Grouping people into 5 clusters}
 \label{fig:bipartite_revolution_source}
 \end{subfigure}
 \begin{subfigure}{.49\textwidth}
 \centering
-\includegraphics[scale=0.4,draft=false]{%
-../../results/american_revolution/american_revolution_dest.pdf}
+%\includegraphics[scale=0.4,draft=false]{%
+%../../results/american_revolution/american_revolution_dest.pdf}
 \caption{Grouping organisations into 2 clusters}
 \label{fig:bipartite_revolution_dest}
 \end{subfigure}
@@ -1508,8 +1508,8 @@ \section{Unicode Languages network} \label{sec:bipartite_languages}
 
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.6, draft=false]{%
-../../results/languages/languages_source_map_clusts.pdf}
+%\includegraphics[scale=0.6, draft=false]{%
+%../../results/languages/languages_source_map_clusts.pdf}
 \caption{Clustering the territories from the Unicode Languages network}
 \label{fig:bipartite_languages_map}
 \end{figure}
@@ -1911,8 +1911,8 @@ \section{Examples}
 %
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.7,draft=false]{%
-../tikz/instance_example_1/instance_example_1.pdf}
+%\includegraphics[scale=0.7,draft=false]{%
+%../tikz/instance_example_1/instance_example_1.pdf}
 \caption{The specified graph $\ca{G}$ and anchored motif $\ca{M}$}
 \label{fig:instance_example_1}
 \end{figure}
@@ -1934,8 +1934,8 @@ \section{Examples}
 %
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.7,draft=false]{%
-../tikz/instance_example_2/instance_example_2.pdf}
+%\includegraphics[scale=0.7,draft=false]{%
+%../tikz/instance_example_2/instance_example_2.pdf}
 \caption{Functional instances $\ca{H}_1,\ca{H}_2$ and $\ca{H}_3$}
 \label{fig:instance_example_2}
 \end{figure}
@@ -1978,7 +1978,7 @@ \section{Examples}
 %
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.7,draft=false]{../tikz/M6/M6.pdf}
+%\includegraphics[scale=0.7,draft=false]{../tikz/M6/M6.pdf}
 \caption{The motif $\ca{M}_6$}
 \label{fig:M6}
 \end{figure}
@@ -2417,8 +2417,8 @@ \section{US map} \label{sec:notes_us_map}
 \vspace*{-0.8cm}
 \begin{figure}[H]
 \centering
-\includegraphics[scale=0.6,draft=false]{%
-../../results/us_migration/us_migration_map_state_names.pdf}
+%\includegraphics[scale=0.6,draft=false]{%
+%../../results/us_migration/us_migration_map_state_names.pdf}
 \vspace*{-0.5cm} \caption{US map with state boundaries and state
 abbreviations} \label{fig:notes_us_map}
 \end{figure}
diff --git a/tests/masters_dissertation_out.tex b/tests/masters_dissertation_out.tex
index 6f2b2a3..a812d10 100644
--- a/tests/masters_dissertation_out.tex
+++ b/tests/masters_dissertation_out.tex
@@ -1,4 +1,4 @@
-\documentclass[12pt]{ociamthesis}
+\documentclass[12pt,draft]{ociamthesis}
 
 %TC:ignore
 
@@ -45,15 +45,15 @@
 \usepackage{subcaption}
 
 % draft options
-\usepackage{ifdraft}
-\ifoptiondraft{
-  \usepackage{draftwatermark}
-  \SetWatermarkText{DRAFT}
-  \SetWatermarkScale{6}
-  \SetWatermarkColor[rgb]{1,0.9,0.9}
-  \usepackage{showframe}
-  \usepackage{layout}
-}{}
+%\usepackage{ifdraft}
+%\ifoptiondraft{
+%\usepackage{draftwatermark}
+%\SetWatermarkText{DRAFT}
+%\SetWatermarkScale{6}
+%\SetWatermarkColor[rgb]{1,0.9,0.9}
+%\usepackage{showframe}
+%\usepackage{layout}
+%}{}
 
 % hyperlinks
 \usepackage[plainpages=false,draft=false
@@ -215,7 +215,7 @@
   \begingroup
   \let\cleardoublepage\relax
   \let\clearpage\relax
-  \printglossary[title=Abbreviations, style=mystyle]
+  %\printglossary[title=Abbreviations, style=mystyle]
   \endgroup
 
 \end{romanpages}
@@ -404,8 +404,8 @@ \section{Graph definitions} \label{sec:graphs_graph_definitions}
 %
 \begin{figure}[H]
   \centering
-  \includegraphics[scale=0.7,draft=false]{%
-  ../tikz/motif_definitions_directed/motif_definitions_directed.pdf}
+  %\includegraphics[scale=0.7,draft=false]{%
+  %../tikz/motif_definitions_directed/motif_definitions_directed.pdf}
   \caption{All simple motifs on at most three vertices}
   \label{fig:motif_definitions_directed}
 \end{figure}
@@ -810,16 +810,16 @@ \subsection{Eigenvector sweep} \label{sec:spectral_sweep}
 \begin{figure}[H]
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../tikz/eigenvector_sweep_network/eigenvector_sweep_network.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../tikz/eigenvector_sweep_network/eigenvector_sweep_network.pdf}
     \caption{A small network}
     \label{fig:eigenvector_sweep_network}
   \end{subfigure}
   %
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/eigenvector_sweep/eigenvector_sweep_scores.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/eigenvector_sweep/eigenvector_sweep_scores.pdf}
     \caption{Sweep profile of the network}
     \label{fig:eigenvector_sweep_profile}
   \end{subfigure}
@@ -1001,8 +1001,8 @@ \subsection{Symmetric two-block DSBMs}
 
 \begin{figure}[H]
   \centering
-  \includegraphics[scale=0.8,draft=false]{%
-  ../tikz/sym_two_block_dsbm/sym_two_block_dsbm.pdf}
+  %\includegraphics[scale=0.8,draft=false]{%
+  %../tikz/sym_two_block_dsbm/sym_two_block_dsbm.pdf}
   \caption{Symmetric two-block DSBM block structure and sparsity matrix}
   \label{fig:sym_two_block_dsbm}
 \end{figure}
@@ -1024,14 +1024,14 @@ \subsection{Symmetric two-block DSBMs}
 \begin{figure}[H]
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/motifsym/motifsym_1.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/motifsym/motifsym_1.pdf}
     \caption{$n=50$, $p=0.3$, $q=0.2$}
   \end{subfigure}
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/motifsym/motifsym_2.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/motifsym/motifsym_2.pdf}
     \caption{$n=100$, $p=0.15$, $q=0.1$}
   \end{subfigure}
   \caption{ARI violin plots for the symmetric two-block DSBM}
@@ -1051,8 +1051,8 @@ \subsection{Asymmetric two-block DSBMs} \label{sec:motif_asymm_dsbms}
 
 \begin{figure}[H]
   \centering
-  \includegraphics[scale=0.8,draft=false]{%
-  ../tikz/asym_two_block_dsbm/asym_two_block_dsbm.pdf}
+  %\includegraphics[scale=0.8,draft=false]{%
+  %../tikz/asym_two_block_dsbm/asym_two_block_dsbm.pdf}
   \caption{Asymmetric two-block DSBM block structure and sparsity matrix}
   \label{fig:asym_two_block_dsbm}
 \end{figure}
@@ -1075,14 +1075,14 @@ \subsection{Asymmetric two-block DSBMs} \label{sec:motif_asymm_dsbms}
 \begin{figure}[H]
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/motifasym/motifasym_1.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/motifasym/motifasym_1.pdf}
     \caption{$n=100$, $p=0.2$, $q_1=0.35$, $q_2=0.05$}
   \end{subfigure}
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/motifasym/motifasym_2.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/motifasym/motifasym_2.pdf}
     \caption{$n=200$, $p=0.15$, $q_1=0.25$, $q_2=0.05$}
   \end{subfigure}
   \caption{ARI violin plots for the asymmetric two-block DSBM}
@@ -1115,15 +1115,15 @@ \section{US Political Blogs network} \label{sec:motif_polblogs}
 \begin{figure}[H]
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/polblogs/polblogs_network.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/polblogs/polblogs_network.pdf}
     \caption{The US Political Blogs network}
     \label{fig:polblogs_network}
   \end{subfigure}
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/polblogs/polblogs_ari_conn.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/polblogs/polblogs_ari_conn.pdf}
     \caption{ARI against $|C|$ across motifs}
     \label{fig:polblogs_ariplot}
   \end{subfigure}
@@ -1143,15 +1143,15 @@ \section{US Political Blogs network} \label{sec:motif_polblogs}
 \begin{figure}[H]
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/polblogs/polblogs_M12_truth.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/polblogs/polblogs_M12_truth.pdf}
     \caption{Colouring by truth label}
     \label{fig:polblogs_embedding_truth}
   \end{subfigure}
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/polblogs/polblogs_M12_clusts.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/polblogs/polblogs_M12_clusts.pdf}
     \caption{Colouring by $k$-means++ cluster}
     \label{fig:polblogs_embedding_kmeans}
   \end{subfigure}
@@ -1192,20 +1192,20 @@ \section{US Migration network} \label{sec:motif_migration}
 \begin{figure}[H]
   \begin{subfigure}{.325\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/us_migration/us_migration_sweep_profile_Ms.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/us_migration/us_migration_sweep_profile_Ms.pdf}
     \caption{$\ca{M}_\mathrm{s}$}
   \end{subfigure}
   \begin{subfigure}{.325\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/us_migration/us_migration_sweep_profile_M6.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/us_migration/us_migration_sweep_profile_M6.pdf}
     \caption{$\ca{M}_6$}
   \end{subfigure}
   \begin{subfigure}{.325\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/us_migration/us_migration_sweep_profile_M9.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/us_migration/us_migration_sweep_profile_M9.pdf}
     \caption{$\ca{M}_9$}
   \end{subfigure}
   \caption{Sweep profiles of the US Migration network}
@@ -1257,7 +1257,7 @@ \section{US Migration network} \label{sec:motif_migration}
     \centering
     \setlength{\tabcolsep}{0em}
     \begin{tabular}{ |c|c|c|c| }
-      \expandableinput ../../results/us_migration/us_migration_table.txt
+      %\expandableinput ../../results/us_migration/us_migration_table.txt
     \end{tabular}
   \end{table}
   \vspace*{-0.5cm}
@@ -1299,7 +1299,7 @@ \subsection{Collider and expander motifs} \label{sec:coll_expa}
 
 \begin{figure}[H]
   \centering
-  \includegraphics[scale=0.8,draft=false]{../tikz/expa_coll/expa_coll.pdf}
+  %\includegraphics[scale=0.8,draft=false]{../tikz/expa_coll/expa_coll.pdf}
   \caption{The collider and expander motifs}
   \label{fig:expa_coll}
 \end{figure}
@@ -1396,8 +1396,8 @@ \section{Bipartite stochastic block models} \label{sec:bipartite_sbms}
 
 \begin{figure}[H]
   \centering
-  \includegraphics[scale=0.8,draft=false]{%
-  ../tikz/bipartite_dsbm/bipartite_dsbm.pdf}
+  %\includegraphics[scale=0.8,draft=false]{%
+  %../tikz/bipartite_dsbm/bipartite_dsbm.pdf}
   \caption{BSBM block structure and sparsity matrix}
   \label{fig:bipartite_bsbm}
 \end{figure}
@@ -1418,14 +1418,14 @@ \section{Bipartite stochastic block models} \label{sec:bipartite_sbms}
 \begin{figure}[H]
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/bipartite/bipartite1.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/bipartite/bipartite1.pdf}
     \caption{$n=100$, $p=0.2$, $q=0.1$}
   \end{subfigure}
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/bipartite/bipartite2.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/bipartite/bipartite2.pdf}
     \caption{$n=200$, $p=0.1$, $q=0.06$}
   \end{subfigure}
   \caption{ARI violin plots for the BSBM}
@@ -1452,15 +1452,15 @@ \section{American Revolution network} \label{sec:bipartite_american_revolution}
 \begin{figure}[H]
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/american_revolution/american_revolution_source.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/american_revolution/american_revolution_source.pdf}
     \caption{Grouping people into 5 clusters}
     \label{fig:bipartite_revolution_source}
   \end{subfigure}
   \begin{subfigure}{.49\textwidth}
     \centering
-    \includegraphics[scale=0.4,draft=false]{%
-    ../../results/american_revolution/american_revolution_dest.pdf}
+    %\includegraphics[scale=0.4,draft=false]{%
+    %../../results/american_revolution/american_revolution_dest.pdf}
     \caption{Grouping organisations into 2 clusters}
     \label{fig:bipartite_revolution_dest}
   \end{subfigure}
@@ -1508,8 +1508,8 @@ \section{Unicode Languages network} \label{sec:bipartite_languages}
 
 \begin{figure}[H]
   \centering
-  \includegraphics[scale=0.6, draft=false]{%
-  ../../results/languages/languages_source_map_clusts.pdf}
+  %\includegraphics[scale=0.6, draft=false]{%
+  %../../results/languages/languages_source_map_clusts.pdf}
   \caption{Clustering the territories from the Unicode Languages network}
   \label{fig:bipartite_languages_map}
 \end{figure}
@@ -1911,8 +1911,8 @@ \section{Examples}
   %
   \begin{figure}[H]
     \centering
-    \includegraphics[scale=0.7,draft=false]{%
-    ../tikz/instance_example_1/instance_example_1.pdf}
+    %\includegraphics[scale=0.7,draft=false]{%
+    %../tikz/instance_example_1/instance_example_1.pdf}
     \caption{The specified graph $\ca{G}$ and anchored motif $\ca{M}$}
     \label{fig:instance_example_1}
   \end{figure}
@@ -1934,8 +1934,8 @@ \section{Examples}
   %
   \begin{figure}[H]
     \centering
-    \includegraphics[scale=0.7,draft=false]{%
-    ../tikz/instance_example_2/instance_example_2.pdf}
+    %\includegraphics[scale=0.7,draft=false]{%
+    %../tikz/instance_example_2/instance_example_2.pdf}
     \caption{Functional instances $\ca{H}_1,\ca{H}_2$ and $\ca{H}_3$}
     \label{fig:instance_example_2}
   \end{figure}
@@ -1978,7 +1978,7 @@ \section{Examples}
   %
   \begin{figure}[H]
     \centering
-    \includegraphics[scale=0.7,draft=false]{../tikz/M6/M6.pdf}
+    %\includegraphics[scale=0.7,draft=false]{../tikz/M6/M6.pdf}
     \caption{The motif $\ca{M}_6$}
     \label{fig:M6}
   \end{figure}
@@ -2417,8 +2417,8 @@ \section{US map} \label{sec:notes_us_map}
 \vspace*{-0.8cm}
 \begin{figure}[H]
   \centering
-  \includegraphics[scale=0.6,draft=false]{%
-  ../../results/us_migration/us_migration_map_state_names.pdf}
+  %\includegraphics[scale=0.6,draft=false]{%
+  %../../results/us_migration/us_migration_map_state_names.pdf}
   \vspace*{-0.5cm} \caption{US map with state boundaries and state
   abbreviations} \label{fig:notes_us_map}
 \end{figure}
diff --git a/tests/readme_in.tex b/tests/readme_in.tex
index 6b67639..5e9202a 100644
--- a/tests/readme_in.tex
+++ b/tests/readme_in.tex
@@ -7,8 +7,8 @@
 over multiple lines
 \end{itemize}
 
-\begin{align}
+\begin{equation}
 E = m c^2
-\end{align}
+\end{equation}
 
 \end{document}
diff --git a/tests/readme_out.tex b/tests/readme_out.tex
index a1531f2..2668698 100644
--- a/tests/readme_out.tex
+++ b/tests/readme_out.tex
@@ -7,8 +7,8 @@
     over multiple lines
 \end{itemize}
 
-\begin{align}
+\begin{equation}
   E = m c^2
-\end{align}
+\end{equation}
 
 \end{document}
diff --git a/tests/wrap_in.tex b/tests/wrap_in.tex
index 21af7e8..0dc42d1 100644
--- a/tests/wrap_in.tex
+++ b/tests/wrap_in.tex
@@ -1,3 +1,7 @@
+\documentclass{article}
+
+\begin{document}
+
 % no comment
 This line is too long because it has more than eighty characters inside it. Therefore it should be split.
 
@@ -24,3 +28,5 @@
 
 % double break after only comment
 % This line is all a long comment. This comment is very long so needs to be split over three lines which is another edge case which should be checked here with all these extra words
+
+\end{document}
diff --git a/tests/wrap_out.tex b/tests/wrap_out.tex
index 19c0a6e..d2c40f7 100644
--- a/tests/wrap_out.tex
+++ b/tests/wrap_out.tex
@@ -1,3 +1,7 @@
+\documentclass{article}
+
+\begin{document}
+
 % no comment
 This line is too long because it has more than eighty characters inside it.
 Therefore it should be split.
@@ -33,3 +37,5 @@
 % This line is all a long comment. This comment is very long so needs to be
 % split over three lines which is another edge case which should be checked
 % here with all these extra words
+
+\end{document}