Update classes.py

warrior bug fixed