Improve prediction