Updates to prediction scripts

Bug fixes to replicate the published work.