Fixed prediction

Increased demo simulation's accuracy