added tests for for_stmt