Added additional parameters to generate neural networks