Added neural networks to the workflow