Added command line option -nl for setting the non-linearity layer.