Basic Neural Network Python Module Definition Architecture