Basic Neural Network Python Code For Prime