Now all working with pytreatments