feat(predictions): performance imporvements for prediction alogrithm