updated the predict atrocities method

fixed bugs in indexing