Added code to generate Neural Network graph.