Fix neural network and command line