Fix 'if' statement-Add TODOs