updating minor change in run_cnn