Basic Neural Network Python Code Tester