incremental changes to lenet.py