Add IfStatement management