Modify definitions

1 files changed, 11 insertions, 7 deletions

diff --git a/definitions.sty b/definitions.sty
index 3651d78..0726ee5 100644
--- a/definitions.sty
+++ b/definitions.sty
@@ -1,19 +1,23 @@
 \usepackage{ifthen}
 
-\newtheorem{definition}{Definition}[chapter]
 \newtheorem{theorem}{Theorem}[chapter]
-\newtheorem{lemma}{Lemma}[chapter]
-\newtheorem{observation}{observation}[chapter]
+\newtheorem{problem}[theorem]{Problem}
+\newtheorem{definition}[theorem]{Definition}
+\newtheorem{lemma}[theorem]{Lemma}
+\newtheorem{observation}[theorem]{Observation}
 
 % #1 (optional): Fixed parameters of the problem
 % #2: Name of query problem
 % #3: The structure to preprocess
 % #4: Queries to handle
 \newcommand{\queryproblem}[4][]{
-    \textsc{#2} \\
-    \ifthenelse{\equal{#1}{}}{}{\textbf{Fixed:} #1 \\}
-    \textbf{Given:} #3 \\
-    \textbf{Queries:} #4
+    \begin{problem}
+        \upshape
+        \textsc{#2} \\
+        \ifthenelse{\equal{#1}{}}{}{\textbf{Fixed:} #1 \\}
+        \textbf{Given:} #3 \\
+        \textbf{Queries:} #4
+    \end{problem}
 }
 
 \newcommand{\qptime}[2]{$\langle$#1, #2$\rangle$}