add prediction module