Update mwb_generator.py