Government invests £4m in AI to help teacher workload