Adding Neural Network code