updated B.py