add predict module