add flat prediction