Feature: Simple prediction algorithm working