def classify_token(w):
    """
    Function for feature augmentation. Improves the network's results by generating
    the information that will be fed into the feature-augmentend embedding layer

    Parameters
    ----------
    w : str
        Word
    
    Returns
    -------
    str
        Class to which a token pertains by basic feature extraction
    """

    suffixes = ["azole", "idine", "amine", "mycin", "xacin", "ostol", "adiol"]
    suffixes_drug = ["ine", "cin", "ium", "ide"]
    suffixes_brand = ["gen"]
    suffixes_group = ["etic","tics", "acid", "tors", "cids"]
    pref_group = ['anta','thia','ace','diur']

    # if w.isupper() or str.lower(w[-3:]) in suffixes_brand: 
    #     return "brand"
    # elif str.lower(w[-5:]) in suffixes or str.lower(w[-3:]) in suffixes_drug: 
    #     return "drug"
    if str.lower(w[-4:]) in suffixes_group or "agent" in str.lower(w) or str.lower(w[:4]) in pref_group:
        return "group"
    else : 
        return "other"