Cleaned up WGGen.py