fixing neural net builder