def p_simple_stmt(p):
    """ simple_stmt : small_stmt
                    | simple_stmt ";" small_stmt """
    if len(p) == 2:
        p[0] = ['STATEMENTS', [p[1]]]
    else:
        p[0] = ['STATEMENTS', p[1][1] + [p[3]]]
