Update struct_improved.py