Mean Field Models with Heterogeneous Agents: Extensions and Learning