--- a/src/ast_rewrite.rs
+++ b/src/ast_rewrite.rs
@@ -40,6 +40,15 @@ impl Visitor<()> for Rewrite {
                     self.visit_statement(statement);
                 }
             },
+            Statement::While(test, body, orelse) => {
+                self.visit_expr(test);
+                for statement in body {
+                    self.visit_statement(statement);
+                }
+                for statement in orelse {
+                    self.visit_statement(statement);
+                }
+            },
             Statement::For(target, iter, body, orelse) => {
                 self.visit_expr(target);
                 self.visit_expr(iter);
@@ -65,6 +74,7 @@ impl Visitor<()> for Rewrite {
             Statement::Expr(expr) => {
                 self.visit_expr(expr);
             },
+            Statement::Break => {},
             Statement::Error => {
                 println!("Statement::Error");
                 panic!()