Update test_control_structures.py