\begin{figure}
    \input{papers/kra/images/phase_space.tex}
    % \begin{minipage}{.45\textwidth}
    %     \input{papers/kra/images/phase_space_small_omega.tex}
    % \end{minipage}
    % \begin{minipage}{.45\textwidth}
    %     \input{papers/kra/images/phase_space_large_omega.tex}
    % \end{minipage}
    % \begin{minipage}[.5\textwidth]
    %     \input{papers/kra/images/phase_space_large_omega.tex}
    % \end{minipage}
\end{figure}