Added flat prediction interface