add prediction
fix prediction