def p_for_stmt(p):
    """for_stmt : FOR id_list IN testlist_star_expr suite
                | FOR "[" id_list "]" IN testlist_star_expr suite """
    if len(p) == 6:
        p[0] = ['FOR', p[2], p[4], p[5]]
    else:
        p[0] = ['FOR', p[3], p[6], p[7]]
