Update weight_problemsII.py