Update wb.py