Update generator3.py

Fixed for reorganization