Adding initial neural network module