Improving touch screens with AI