clean treat_assignment and generate variables