add feature prediction