Added rudimentary neural network logic, need to wrap into a class/function