add for loop to control_flow test