added display option to neural net