Basic Neural Network Python Notebook