change algorythm of prediction
Merge branch 'work'