Fixing the exclusion of variables
Fixes the problem with shape of first layer not matching the reduced size of inputs when excluding variables
Fixes the problem with shape of first layer not matching the reduced size of inputs when excluding variables