Draftmode with comment options

Chapter1/chapter1.tex

@@ -29,6 +29,7 @@ known as the \textbf{energy-mass-momentum} relation as an in-line equation.
 
 A {\em \LaTeX{} class file}\index{\LaTeX{} class file@LaTeX class file} is a file, which holds style information for a particular \LaTeX{}.
 
+
 \begin{align}
 CIF: \hspace*{5mm}F_0^j(a) = \frac{1}{2\pi \iota} \oint_{\gamma} \frac{F_0^j(z)}{z - a} dz
 \end{align}

Classes/PhDThesisPSnPDF.cls

@@ -209,7 +209,7 @@ footnotes.}
 \DeclareOption{twoside}{\PassOptionsToClass{\CurrentOption}{book}}%
 
 % Draft Mode
-\DeclareOption{draft}{\PassOptionsToClass{\CurrentOption}{book} 
+\DeclareOption{draft}{\PassOptionsToClass{\CurrentOption}{book}
 \ClassWarning{PhDThesisPSnPDF}{Draft is ON. You can use `draftmode' instead,
   which adds watermark with timestamp and line numbering. Don't forget to turn 
   `draft' off for your final version.}}%

Preamble/preamble.tex

@@ -182,9 +182,13 @@
 %% Todo notes functionality
 %% Uncomment the following lines to have todonotes.
 
-%\usepackage[colorinlistoftodos]{todonotes}
-%\newcommand{\mynote}[1]{\todo[author=kks32,size=\small,inline,color=green!40]{#1}}
-
+%\ifsetDraft
+%	\usepackage[colorinlistoftodos]{todonotes}
+%	\newcommand{\mynote}[1]{\todo[author=kks32,size=\small,inline,color=green!40]{#1}}
+%\else
+%	\newcommand{\mynote}[1]{}
+%	\newcommand{\listoftodos}{}
+%\fi
 
 % Example todo: \mynote{Hey! I have a note}