study_arm function

Indicate study arm variable in formula