Update artTRAIN.py