\begin{problem} \end{problem}
should be replaced with \begin{task} \end{task}
\begin{task} \end{task}