Update mainDL.py

did some testing