implementing prediction module