<s>[INST] def sum_of_squares(n):
    """Write a function that takes a string of parentheses and/or letters as input and returns the total number of 'a's in each set of balanced parentheses. 

    For example, given the string "(a)(a)aa" the function should return [1, 1, 2]. 

    You can assume that the parentheses are properly balanced."""
def sum_of_squares(n):
    pass[/INST]