Added features distribution checks; Added id_varnames to be retained in train, validate & test datasets