Update dnn.py