Added simple neural net module