Work for AY250 Python class