Building a Neural Network tool in Python