Fixed Python 2 incompatibility in NN notebook