Update mainb.py