Added working full prediction (though very slow)