Improved prediction configuration