added display argument to run the new function
added display argument to run the new function

Warvito/pydeeplearn,snurkabill/pydeeplearn,Warvito/pydeeplearn,mihaelacr/pydeeplearn,snurkabill/pydeeplearn