\begin{figure}[!htbp]
    \centering
    \begin{subfigure}{.5\textwidth}
      \centering
      \includegraphics[scale=0.5]{compare_av_reward_episode.png}
      \caption{Average reward per episode}
      \label{fig:sub13}
    \end{subfigure}%
    \begin{subfigure}{.5\textwidth}
      \centering
      \includegraphics[scale=0.5]{compare_av_reward_state.png}
      \caption{Average reward per state.}
      \label{fig:sub14}
    \end{subfigure}
    \caption{Comparing SARSA and Q-Learn Algorithms}
    \label{fig:test5}
\end{figure}