Update ChooseClass.py