I wonder if some of the issue is shortened attention span.
I swear those devices rewire our brains, and it's probably worse in younger children. I'd start with more reading to your son, and if you need a break try to find some old Sesame Street episodes or something for him to watch.