add reqiured data
Update psottraining.py