Improved prediction example