Updating feature gen in split models