Updated the for_loop model. Replaced Integer with variable