Improved the prediction script
Merge pull request #1066 from denadai2/master

Improved the prediction script