import re


def get_treatment(s):
    """Get the type of treatment from filename"""
    temp = re.split("_|\.", s.lower())
    if "saline" in temp:
        return "control"
    elif "muscimol" in temp or "musc" in temp:
        return "muscimol"