Update class.py