Add win prediction feature