modified python file to adjust number of neurons for input