Update round1/strength.py