Make the inputs more flexible.  Automatically determine the size of the first layer.