US general says Iraq tackling militia infiltration