Add a basic method to plot the neural networks
Add more information to plot_neural_net