Basic Neural Network Python Programming