Rewrite size classifier to use linear flattened structure