Fixes to neural network notebook