Krusell Smith almost working. Need to speed up solving and fix the simulation phase.