Basic Neural Network Python Example Code