updated arm_assist simulation code