Update class_plateau.py