NN input generation works