Basic Neural Network Python Code For Prime Number