Added own prediction tweaks