add a few more tests and some cleanup
some cleanup of ForallStmt stuff