Improving AI's ability to identify students who need help