Add flat_mode