Remove pytreatments
Add pytreatments from correct place