Flat model architecture