Put parameters for NN in separate class.