Basic Neural Network Code In Python For Prime